from langchain.prompts import ChatPromptTemplate
from tool import langchain_client
from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory
if __name__ == '__main__':
    memory = ConversationBufferMemory()
    # 新建一个 ConversationChain Class 实例
    # verbose参数设置为True时，程序会输出更详细的信息，以提供更多的调试或运行时信息。
    # 相反，当将verbose参数设置为False时，程序会以更简洁的方式运行，只输出关键的信息。
    conversation = ConversationChain(llm=langchain_client, memory = memory, verbose=True )

    response = conversation.predict(input="你好, 我叫皮皮鲁")
    print(response)

    response1 = conversation.predict(input="1+1等于多少?")
    print(response1)

    response2 = conversation.predict(input="我叫什么名字?")
    print(response2)

    print(memory.buffer)

    print(memory.load_memory_variables({}))

